#pragma once
#ifdef RCT_NEW_ARCH_ENABLED

#include <react/renderer/components/root/RootShadowNode.h>
#include <react/renderer/core/ShadowNode.h>

#include <unordered_map>
#include <utility>

using namespace facebook;
using namespace react;

namespace reanimated {

class PropsRegistry {
  public:
    std::lock_guard<std::mutex> createLock() const;
    // returns a lock you need to hold when calling any of the methods below

    void update(const ShadowNode::Shared &shadowNode, folly::dynamic &&props);

    void for_each(std::function<void(
                      const ShadowNodeFamily &family,
                      const folly::dynamic &props)> callback) const;

    void remove(const Tag tag);

    void pleaseSkipReanimatedCommit()
    {
        shouldReanimatedSkipCommit_ = true;
    }

    bool shouldReanimatedSkipCommit()
    {
#if REACT_NATIVE_MINOR_VERSION >= 73
        return shouldReanimatedSkipCommit_;
#else
        return shouldReanimatedSkipCommit_.exchange(false);
#endif
    }

#if REACT_NATIVE_MINOR_VERSION >= 73
    void resetReanimatedSkipCommitFlag()
    {
        shouldReanimatedSkipCommit_ = false;
    }
#endif

  private:
    std::unordered_map<Tag, std::pair<ShadowNode::Shared, folly::dynamic>> map_;

    mutable std::mutex mutex_; // Protects `map_`.

    std::atomic<bool> shouldReanimatedSkipCommit_;
};

} // namespace reanimated

#endif // RCT_NEW_ARCH_ENABLED
